from bs4 import BeautifulSoup
import json

class DataProcessor:
    def __init__(self):
        pass

    def process_response(self, response):
        """处理响应数据，基于内容类型"""
        content_type = response.headers.get('Content-Type', '')

        if 'application/json' in content_type:
            return self._process_json(response.json())
        elif 'text/html' in content_type:
            return self._process_html(response.text)
        else:
            raise ValueError(f"未处理的内容类型: {content_type}")

    def _process_json(self, data):
        """处理JSON数据"""
        # 这里根据需要处理JSON数据
        return json.dumps(data)  # 示例：转换为JSON字符串

    def _process_html(self, html_content):
        """处理HTML内容"""
        # 使用BeautifulSoup解析HTML
        soup = BeautifulSoup(html_content, 'html.parser')
        # 根据需要处理HTML，例如提取特定元素
        return str(soup)

# 使用示例
# data_processor = DataProcessor()
# response = ... # 从LeetCode或其他网站获取的响应
# processed_data = data_processor.process_response(response)
